Changed challenge 5 to use arguments rather than stdin