adding more class code