Assignment 1 for COMP397