DOC: Cleaned up comments