Some UI work on iOS app