Fixed search highlighting bug. Some minor UI changes as well.
Updated the screenshots to show the new UI design.