Adding speed testing code