Add initial code for updating ML