Update baton.cpp