Malala to return to Pak after treatment