fixing code formatting