add imports to example to make complete program