Cleaned up the code a little bit.
Python the hard way work
moar python!