added another if statment