Finished AI, needs testing