Initial code added in ViewController.swift