basic coding exercise in idris