Remove old test file.
Improve docstring
PEP8 changes