Fixed the issues with remove()
Continue implementing rest of functions and more tests