Arduino Nano communication problem

One more question. If I want to restore USB programming ability - is it enough to just add:

  pinMode(0, INPUT);

I want to remove 2s pause at the start, but want to enable programming by external button.