Update proma.py