Fixed broken test puzzle :)
Fixed the main file with prettier printing (<3) for the input file.