Updates for assign, if, while. Need to fix the positive while case.