Add a class diagram, and remove some necessary methods.