More updates to class dia. Added more state diagrams