added some code for that extension