Update comments for more clarification