cleanup the code.