Moved the OK Button listener