That actually made everything a whole lost easier by adding an override func of viewDidAppear. Thank you