Add Model to load functionality