Add do/while to type inference