Using prop instead of attr. Fixing bug
remove button