Update wx531.py

added the top two lines and fixed up the 1-rep max input boxes.