Added missing kwargs to save file