Added UML class diagram. Cleaned up more code.