Add watchdog functionality