Adding discount logic