Update rosalind_prtm.py