Cleaned up pull tests based on pull request feedback.
Added some docstrings on main files.