cleaned code up a bit by meeting PEP 8 guidelines