Change drop off a little bit to fix a bug.

Make it two stages. In the first stage, drop all elements which
are outside the specified maximum bandwidth. Check if we have
reached the budget. If not, in the second stage, continue dropping
until we reach the main diagonal or reach the budget.