Add possibility to work with different categories