Add stub for training