Improve JSDoc comments