One suggestion: adding a button to the challenges to restart the battle would be great, so you could repeat the battle without having to return to the challenge's initial screen.
