Add edit note feature