Fixed some python 2.7 syntax