Added logic to select option based on user provided argument