Added data generator mode

As requested.