day1 python code