Add checkboxes for filtration
Add tests