Added a function

Just added 'return' to dialogue script so that both enter buttons work