Add method to save file