Updated the relevant comments