Add documentation on usage
Update test file (wasn't running previously)