Issue #80, add UML