Add heartbeat check