Added example to readme and cleaned up imports