Added plotting script!