updating ML code