Update UML doc to reflect Mock