Completed challenge, added docstrings