Passing all tests now
Added docstring
Added some comments