I haven't tried on that chip but generally the registers have the same names as are given in the datasheet (or the datasheet would be very confusing). You basically need to see what the original code does (eg. vis-a-vis BODS) by looking at the Atmega328 datasheet, and then see what bits in what register accomplish the same thing on the new processor.