Very minor code changes

Very minor import and storyboard changes