add test.js for mocha
add if-else statement
add main.js