Update plotting script for SIT151 C TP2