Edit functionality added