Add note taking functionality