Update wavePropagation.js