Not working yet, but added a goal and hopefully frame distillation