Small improvements to arma.py