Add plotting script