Added input to main.c implementation.