Added note saving back, please let me know if you see any further issues