Adding in models and load functions