Your program is very close to the above commenter's program. You could have picked the identical program which is exactly what the proof does.