Armstring Numbers easy challenge solved in python