Setting up implemintation in the ViewController