Added some docstrings, still some work todo