Cleaned up example a bit.