上のプログラムにさらにアレンジしました。こんな感じです。