COMP2301-assignment-2