output was pretty bad. Identation of the prints got wrong at some point. Fixed now