Unit class
Merge pull request #6 from wynestic/master

Unit class