Code cleanup decorator