Add ability to set a mode