added one line if statement