Prototyping, some functionality with buttons, working on main data class