How to rewrite it to functional style