adding the data class and controller classes