Added basic search feature (super ugly)