changed end battle screen