Improve view post functionality