Complete unit tests and model setup