clean up example code and include test output