Updated readme with info about the model