Changing mode with buttons now works!