Updating class to save file to device