Perform code cleanup