Mini code cleanup