Create armstrong.py