flat ui redesign, added edit button functionality