Doc fix - adding imports at the top of example section