implemented search functionality, changed a bit of UI