Add comment and fold test