- Added data class and connection setup