Class diagram modifyed