Create data.py

First attempt at data class