Added a few missing lines