Imran Khan, Bajwa take stock of Balochistan situation