Update battleship.rb