Changed the UI of the remove function