Added testing for the remove functionality.