Assignment 2 for COMP3421