Remove Model from main.py

Attempting to split the files up for readbility