changed so that we can run it on different categories