Began tidying up code and adding docstrings