working on making proper unit tests in tests.py, made some code more pep 8 compliant.