Revised UI design for the wireframe