Added some doctrings, and modified docstring tests to make it work