minor clean up added some comments