Updated to just close the register box on devices with small screens