Movement highlighting fixed! Now takes in the 3 unit types as input.