[TASK] Small changes to documentation extension