Added Done button to edit layout