Extending the battleship project I learned on CA