Class 5: Creating the graph network