Fixed Remove Function, correct output