Same code but wave motion instead of radial