Working Python version

Hacked on Hackanoon #7.