Add search controller functionality