Python Graph Plot Code