Finished assignment but remove button not working