My first Python program: a game of battleship