Changing wub logic a bit and stuff.