Add handle_connect_message implementation