Add support for changing focus in max layout