Adding plotting script