Adding dialog for registering treatment