Adjusting input for 5th challenge