Stay diversified to prepare for any more volatility to come, says Jim Cramer