How to rewrite the if else statements into reusable code using javascript