updated code formatting