Updated README and fix on models