improve extension generator