Place model parameters in separate file

Currently all the model parameters are contained in main.py. These should be moved into a separate file.