Add the right align for flax.Label