add classes to create objects on data.py