Improve handling of audio focus and media buttons