cleanup for code main class