Cleaned up code for pep8 complience.