Add support for mode commands