Code for FYP student 1