quick code cleanup