COMP2502-assignment-2