Cleanup some code.