Added in save() on edit. Also some minor UI tweaks.