Shastri believed Virat Kohli could have played 2-3 more years of Test cricket