Set code language to Python