New Flax-based "Hello World" example