More tweaks to code sandbox