Added remove container function