Add heartbeat to prevent websocket disconnects