Fixed shit :D

Signed-off-by: retep998 <retep998@verizon.net>
Merge pull request #15 from retep998/master

Hopefully this fixes any issues...