Updated ViewController to auto add in method proteies