Programming problem for week 3 of UIT