Fixed view controller so that it isn't all hacky for getting the images