Added test on Ozone dataset