Adding Python 3 compatibility