Problem5: python style
Problem5: started haskell solution