Currently, the model to be saved has to be manually selected and copy-pasted separately. Need to improve that piece of code.