Changing code to pep 8 convention