Change Code to Class