Added functionality for search, iteration 2 complete