small modifications to code formatting