Updated readme and added class diagram