Update 01.py

if文をスマートにしました！