Python: Finished battleship app