Added code for GPU registration and client reconnection