As there are now 4 categories, we should think about refactoring this part into 4 different classes.