trying to add if statment