Liten ändring i main scala filen.

Bara en liten editering.