Updates according to latest comments on the FLAVA PR
Update config
Initial model version
initial commit
