Add spec for cleanup method