Refactored the unit class. Added a builder class for the unit class. Also, added a UnitTest to the project. LMAO