Improved documentation for write_atomic_weights.py