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.