Update arsa.py

Fixed issue #5 - sorry about that!