Refactor extension

[m-mujica]

Split the main module, there is no reason to have all the code in a single file/module.