Add diffusion to bouncing ball data generator

[David Sandberg]

Diffusion is used in the paper Learning and Querying Fast Generative Models for Reinforcement Learning to be able to show the advantage of the stochastic state space model family.