bit of code formatting improvements