Improved test code for prettiness