Implement going back to improve view