Fix background.
Refactor, better weight selection.