cleaning up example code