Improve extension example