COMP1501-assignment-2