Successfully modified file: /home/Messier82/projects/service-book/app/src/androidTest/java/com/yorvana/testsupport/DemoMode.kt (1 replacements). Here is the updated code:
...
        runShell("am broadcast -a com.android.systemui.demo -e command network -e mobile show -e datatype none -e level 4")
        runShell("am broadcast -a com.android.systemui.demo -e command notifications -e visible false")
    }

    fun exit() {
        runShell("settings put secure navigation_mode 0") // Reset to 3-button Navigation
        runShell("am broadcast -a com.android.systemui.demo -e command exit")
    }

    fun ensureOutputDirectory() {
        runShell("mkdir -p ${ScreenshotConstants.DEVICE_SCREENSHOT_DIR}")
...