HKUST COMP 5421 project 4