AI not working properlly, need to work on it