World No.1 Aryna Sabalenka is rediscovering her best form in California, scoring a straight-sets win as she moves on from her loss in the Australian Open final.