Code written as part of AI project