Updated model function and README.md