Add watchdog feature