Update snr_comm.py