Add watchdog support