Add test for adding new word, edited some files to match pep8