Working on hero screen restriction