Did a small modification to the view class