Update armstrong.java