Replaced CollectionViewController with ViewController