Fix example and remove extraneous import.