Refactor setting stage for menu styling stories.

Change-Id: I4204465f3f59759b889a2cc18a8ec136bdb8ca97