Added save functionality to classification