Refined treatment plan functionality. Added update usages.