Generate simple pictures from python code