Solves some compile errors
Merge pull request #11 from Nyeriah/pr

Solves some compile errors