add changes to fix remove button function
Merge pull request #5 from aphuong/master

add changes to fix remove button function