Added data.py (generated data, but real problems)