Python based battleship game