Homework project for Introduction  Haskell