updated eda.py