Add error handling in driver, optional  size argument for search