Add battle script
Add checks for valid input