fix bug & add button action