Back to ai assignment 4