COMP6231 project