Transitional. Should find a way to actually take a model as an argument