Add functionality for tooth treatment