Add enter/exit method for with/as statement