Added new weapon and the restart button