code cleanup (same output)