Penalty Unit feature added

The penalty unit feature is added as requested in the requirement.