excuse my attempt at explaining my problem.
Declaration: im working in google colab.
My code bloc essentially returns key values based on conditions of the input a
