Adding additional class code