Changed the import in the example so it's easier to get started

Signed-off-by: Jacky Boen <aqiank@gmail.com>