added remove functionality but flawed