Styling changes to make it work with the edit function