I am not following.
You want #1 to work like #2?