Added option to turn unit on and off