Updated UI to match desktop site