Added comments where they were missing