Though he has worked on his batting, now his bat angle coming down straight rather than at an angle from Australia tour and also has less of a front foot lunge trigger..
He needs to work on his attitude.. Rest he will learn automatically as he moves on and by playing few seasons of first class cricket.