Modify the operator>> to work with k=5. We need to change the
data generation format later and consider editing this again.