Adds docstrings
Refines the code
Disables some pylint messages in tests