Add heartbeat function