Improve photo edit layout