Needed to adapt a little to make it work with python3.6 and tensorflow (imports/print syntax corrections).