Trying to make it work with storyboard