add a if statement