Shocked that South Africa didn't go for win: Kohli