Fix to use displayFeatureId

Currently the button state change depending on the task change on the default display i.e display 0. But there is a possibility that multiple display areas are hosting different application on display 0. The state should depend on DefaultTaskContainer and not display ID.

Bug: 217968579
Test: Manual
Change-Id: Ie980ef6d0e2058e9be1b961ad5469fdc245d2b55