Added python 3 syntax