amended the initial migration file to set up the models