Updates to stage4, stage2 unit test, compiler unit tests