Added input argument