First draft done. Python file to do graph