afair, this is already like that.

when your phone is locked and you receive a call, the screen will show you the normal caller-screen.
after taking the call and then ending it, the phone-app closes and locks the screen again.