taking this code from sentdex tensorflow tutorials - View it on GitHub