finished adding 'remove' function to mock