Edit functionality added.