Improved example usage code in docs