added feature and steps for category regression on the correct branch