Yes that's sounds much better and more enjoyable and forgiving for the player. I would done something similar in my game but being a platformer that takes you back to the beginning of the stated level their no real need because they also rather short.