Add some comments left out