modificate add layout and button