Added healing functionality