Converted ViewController from Obj-C to Swift