Setup data loading in include all of the datapoints
Fix imports in features.py
Restructure example.py to make it cleaner