added imports in code snippets