Style fixes.
Refactor, prevent actions after battle end