World No.1 Aryna Sabalenka got revenge on Madison Keys, who beat her in the Australian Open final last year