Olá,
Relembrando esqueleto de cada EP fornecido por nós é livre para ser modificado. Vocês podem importar outras bibliotecas (que sejam padrão do Java ou estejam no algs4), podem criar métodos privados novos, podem fazer o que quiserem na main. Só não podem modificar a assinatura dos métodos públicos.