Line Graph Python Code