Clean up the models cause I'm an idiot