Added better AI