Add compatibility with mage setup.

Add the two blocks for prices from the mage setup extension