Update extraction code