Update uml diagram to fit actual implementation