update plotting script

fix some of the imports too
rename plotting script