Updated Plan for AI

Small, just updated function that I wrote