Cleanup
Fix call to save_file