Made code neater by condensing

changed else: if: to elif:
Merge pull request #10 from anthonyhlcheng/patch-1

Made code neater by condensing