More time steps added
Modified Jack's example