Convinced it defo works, now for refactor
Removed all printlns, still passes tests