Finally added all the Unit classes... They are all exactly the same except for the image that is loaded on them. Will be changed in the future