Extending the battleship project from CA