Reigning Australian Open champion Madison Keys has flagged some additions to her game as she returns to Melbourne Park.