Entered initial functional code to ViewController.swift