Added more functionality to Save to Flash