Added more asserts to the test and modfied the other files to match pep8