Added main method to graphs.py