MX28: Shuffle around the power management code