Update DataManager.py