Added variable for type B unit on line 12