A minor update in UI and screenshot update