Add battle animations code