Fixed documentation according to review