Add optional flag to inputs