Made the code nicer
Merge pull request #2 from chase4926/patch-1

Made the code nicer