Edit class structure