Fixes code_jail imports and draganddrop yaml

@lyla please review, let's merge it before demo.