Try to add if statement