Move model out of test file into own file.