AI4341_assignment1_team2