Updated the code to be more PEP8 friendly. Personal preference.