Small improvements in test example