Create HW2_madelung.py