Update control scripts to use Buttons