Added module registration code