Filled the 3 classes up a bit and got rid of the errors. Next step is to move the input code from UnitController