Small upgrades. Time to test