update codecademy battleship example