some more comments and uml