setting up the models file, needs work