Changed keep_alive to use the websocket's Heartbeat class