Tweak hw04 tree printing implementation, remove unneeded import