Add variables and initial class code