Add a helper method to the functional tests
Add initial model + tests