improve code with class on scratch02