Remove button works fine too