Fix for python34 users

I am a python34 user and the code wasnt working for me so i fixed it and
added a little touch to it. Requires python34