Also, can you please add a docstring that documents the class.