A little prettier solution