I am also doing the same in this module. Can you refer the code in the module.