Updated control script