[Issue #209][WIP] Fixed UML