add for.s and while.s