Implementation and modification of a three-dimensional radiation stress formulation for surf zone and rip-current applications