Attempt at fixing pretty_print, doesn't work great for dim > 3
Slight improvement in pretty_print