Added comments that were missing