Cleaning up code and updating docs