move graph.py into __init__.py