Small updates to features classes on UML diagram