Getting an error TypeError: cannot unpack non-iterable float object