forgot to add some files to the model