Add a test model function