Add missing imports
Change-Id: Ia8ca0b20c313310f127730d45276ce226d50f34a

Credit @aaronpoweruser
https://github.com/ParanoidAndroid/android_frameworks_base/commit/8318d4bd86ea4c7bd50f88e4b462737fb19c2835