Added code for the search class