Nice puzzle.
There is another way for black to solve this. 4.Nxe5 Bxf2+ 5.Kxf2 Nxe5!
Thanks for ur comment