Remove Function completed

Added some code now it removes. Tests for remove() function as well.