woops :) adding class diagram