moved the model to the top-level