Move code cell config earlier, closes #257