update code with missing lines