Removed test code from the module