Added missing classes to deployment
Updated test to assert for correct number of results