Changed it to python 3x syntax