Updates class diagram