add uml diagram to docs
add changes from docs