Added body count input and restart button