adding if else

add if else