Updated class diagram

fleshed out class diagram