renamed the main file
Little refactoring and added a load function to add external models