Added input parameters to main