Add initial ai funtionality