tweak to answer02.c