Corrected documentation of arms.predict function