add save
fix save
add save in main