Update map screen styles and add UX