Worked on it a bit more, put a loop on the battle