Add price model and other modifications