Added the remove function in init