#3 change class Graph