Added feature to save to local storage