Stango updated the code for problem4 (together with the input file)