Minor changes
Adds the ability to finetune a trained model