add code for extend feature