code cleanup (makes code more pep8)