Update Locomotive.java

Getting functionality working