Update graphtest3.py

Final implementation, working