updates simple arith test to represent valid program
minor ast move arounds