Fix loading issues.
Merge pull request #3 from nanliu/master

Fix puppet lib loading issues
Write unit tests