improve code formatting