working on Remove() and logic errors