Mini refactor btn styles + edit icon