small changes
fix
Merge pull request #4 from MardonHH/patch-1

small changes