Modifying UI slightly.
Adding updated screenshot.