removed unnecessary graph class