Finished my remove method, found a small logic error where it wouldn't function as I thought!