add click listener for return button