Changes to animation js