Added some missing comments