Update bai1.cpp