Price calculation working, but still not the good one