annotate unzip foldr impl and clean main.hs