Add battle state creation