setting up for python plots