Corrección de errores en el código

Agregaron los imports que faltaban