Modelling Treatment Mechanism for Stratified Medicine in Randomised Clinical Trials