Is codex working well with python notebooks?