Locomotive classification