clean up
small clean up