Turn based battle game written in Python