- minor prettying up of code