Small adjustments to viewcontroller