Final project for coursera getting data class