Fixing the names in the treatment control plot