Add test model to generator