Example Python Code For Plotting A 2p Orbital