Kick ass! change to model diagram