improving code formatting