Updated class diagrams to include methods