Tidied docstrings & removed debugging prints