Added missing imports to example from previous commit.