Updated class diagram

still need:
- help button shows nothing
- test functions
- clean up of UI