Slight updates to comments