Add entity framework and add data class