Añadiendo clase para las pruebas unitarias