Added more functions to further AI behavior