first working solution
add generator