Haskell code for learning physics