Updated it to IF statement

Created a separate else statement for future reference. WebStorm editor
on contains IF statement