add if statment