Final submission for AI project