take 3 on the categorical. fixing label issues