Add class diagram
Add better comments