Python implementation of battleship