minor update, code clean up.