Add null check, tweak doc