Adding returns to the distribution generation functions