updated layout to handle small-screen better