Edited class diagram