Make code more pythonic
Start using docopt