#Minor Change

Function to Save and Load model added