hlint suggestions
added necessary import statement