code cleanup-2