Fixing code formatting