Fixed a remove button, now it works with jQuery function, #3