Added ability to save classified data to file