hopefully added the model and data class error free