UI improvements in storyboard