Add missing lines

And typo fix