Add option for wave speed