First py script - Battleship