Add support for key toggles