AI safety enhacement