made changes to address comments and hopefully satisfy the pep8 bot