More speed and code size improvements

Thanks! Luckily it's a small problem - it's literally just that in this case it assumed there would always be an expression after the return if it was the first statement in the function. I'll try and get a fix in for that now.

If you instead did function test(){1;return;} then that should work fine.