Added the return type for remove

Still need to debug