Add functionality to the go button