Trying out a way to make buttons follow category with less code