Changed everything, also AI on button click