python plotting script
updated script