Added getters/setters and improved Unit class. Added Main class to run Unit class.