Tidying up pretty_print