Working on separate plotting script