Added variables on lines 12 and 13 for Type B unit