Added display change functionality