Revert "Remove model"

This reverts commit 3d4ba2119b17bf7cb04aecb1a6a50fb4ff5ec90a.
Update test case