added imports to example usage