got python plotting working