Added a remove function