feat: improve support for importing stdlib in python code blocks

Signed-off-by: Nick Mitchell <[email protected]>

Author: starpit