now it's shiny

Just made everything a bit nicer so that it fits pep8.