Modify the index.js

* Add test function