Improved the session search feature