Refactored total price calculation a bit