Move plot options to python