Clean up imports in example

Just a simple cleanup