Add note editing feature