Update code for battlefield