Quick code cleanup.