Made work with python 2.7