change diagram with new methods