Added Code for remove function