adding example output and cleaning up imports