adding more battle functionality