Style changes to fit in better with pep 8