First homework of Scala day 2