add addtional if

add addtional varaible