Added class diagram

Resolves #11