Why does my Python code give a float object is not iterable error