Updating Battle class