Add save functionality. Update README.