Graph Code In Python