Some last second changes to the python script

Tried to make it flow better and more readable