updated docs + uml