Added ability to search by category