Remove flash element and modify the layout