Get return button working