Refactor edit button to use css