Added the remove function