помогите решить задачу на ветвление в python3.3.

import math
WEIGHT=51
COST=0
if WEIGHT <= 50:
COST=10
else:
COST=10+(5*math.ceil((WEIGHT-50)/50))
m10 = 0
m4 = 0
m1 = 0
if COST >= 10:
m10, COST = divmod(COST,10)
elif COST >= 10:
m4, COST = divmod(COST,4)
else:
m1 = COST
п. с. режет отступы. вообщем думаю поправите