Add class diagram and update README