Updated the module and files added testing as well.