Victoria Azarenka focuses on returning a serve