Changed stuff

Loading data from data.py