Moved code into ViewController