Add in search code