Working search and better styling