Updated vms class diagram