Virat kohli back in form