managed to get remove method working somewhat