Python scripts for CSI2 assignment