saving and loading the model seems to work fine now, still need to make the file extensions checked properly