slight clean up
Merge pull request #1 from kennethklee/fix/cleanup

slight clean up