added support for go back and edit button