Add better calculation to total price