Added imports in readme code sample