Improved user experience. Used pickle to store data