Add stub for remove method