Assignment 2 for COMP125