clean up

code clean up