Added SelectedCategory class. And it works!