Class diagram

I added the class diagram