Add Unit Test Case for search function

Test to check if search function works properly
Updated Unit Test

Change code to recognise the "search" command