Fixed Entity AI Closes #2