Adjusting input for 5th challenge pt2