Correcting code with advice in the PR comments. Added updated storyboard.